from iop import Message
from dataclasses import dataclass
import pandas as pd

@dataclass
class FormationRequest(Message):
    id: int = 0
    BMI: int = ''
    Smoker: bool = ''
    Age: int = ''
    Sex: str = ''

@dataclass
class TrainingInsertRequest(Message):
    BMI: int = ''
    Smoker: bool = ''
    Age: int = ''
    Sex: str = ''

@dataclass
class GeneralResponse(Message):
    status: str = ''
    data: str = ''

@dataclass
class DataFrameRequest(Message):
    data: pd.DataFrame =None